git+https://github.com/davidbau/baukit
accelerate
dataclasses-json
huggingface-hub
invoke
jupyter
jupyter_http_over_ws
matplotlib
numpy
protobuf==3.20.0
requests
seaborn
sentencepiece
tqdm
transformers
torch
plotly

# dev dependencies
black
isort
mypy
pytest
types-requests
